from langdev_helper.llm.lcex import llm_lcex as llm

from langchain_core.prompts import PromptTemplate
from langchain_core.output_parsers import StrOutputParser

prompt_template = '''
作为专业的内容创作者，请围绕[{subject}]写一篇文章。要求：
1. 开头要用故事/数据/观点制造冲突
2. 内容要有具体数据和案例支撑
3. 分点论述，每点都要有实操建议
4. 语言要接地气，像朋友聊天
5. 适当加入自己的观点和思考
6. 结尾要有明确的行动指引
7. 设置2-3个互动引导点
8. 文章结构要清晰，重点突出
'''

prompt = PromptTemplate(
    template=prompt_template, 
    input_variables=["subject"], 
    partial_variables={"foo": "foo"}
)

chain = prompt | llm | StrOutputParser()
resp = chain.invoke({'subject': 'AI赋能教育：如何利用人工智能提高孩子学习效率？'})

print(resp)


